from configs.default import Config


class PrototypeConfig(Config):
    # 原型学习配置
    use_prototype = True
    prototype_dim = 128  # 原型特征维度
    prototype_loss_weight = 0.5  # 原型损失权重
    temperature = 0.1  # 对比损失温度系数

    # 早停配置
    early_stopping_patience = 5
    early_stopping_delta = 0.001

    # 新增GAN配置
    use_gan_augment = True
    gan_augment_ratio = 2.0  # 对少数类样本的增强倍数
    gan_num_epochs = 50

    # 类别权重配置
    class_weights = [1.0, 1.0, 1.0, 1.0, 1.0,
                     1.0, 1.2, 1.2, 1.2, 1.5,
                     1.8]  # fall类别权重设为3